package com.w.picture.api.imagesrarch.model.sub;

import com.w.picture.exception.BusinessException;
import com.w.picture.exception.ErrorCode;
import lombok.extern.slf4j.Slf4j;
import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;
import org.jsoup.nodes.Element;
import org.jsoup.select.Elements;

import java.util.regex.Matcher;
import java.util.regex.Pattern;

/**
 * 获取图片列表页面地址
 * 通过jsoup爬取html页面,提取其中包含firstUrl的JavaScript脚本,并返回图片列表的页面地址
 *
 * @author wzp
 * @since 2025/1/23
 */
@Slf4j
public class GetImageFirstUrlApi {

    /**
     * 获取图片列表页面地址
     *
     * @param url
     * @return
     */
    public static String getImageFirstUrl(String url) {
        try {
            // 使用 Jsoup 获取 HTML 内容
            Document document = Jsoup.connect(url)
                    .timeout(5000)
                    .get();

            // 获取所有 <script> 标签
            Elements scriptElements = document.getElementsByTag("script");

            // 遍历找到包含 `firstUrl` 的脚本内容
            for (Element script : scriptElements) {
                String scriptContent = script.html();
                if (scriptContent.contains("\"firstUrl\"")) {
                    // 正则表达式提取 firstUrl 的值
                    // 这段代码中使用的正则表达式是 "\"firstUrl\"\\s*:\\s*\"(.*?)\""，下面我将为您解释这个正则表达式的各个部分：
                    //
                    //\"firstUrl\"：这部分匹配字符串中的 "firstUrl"。由于双引号在Java字符串中是转义字符，因此需要用反斜杠 \ 来转义它，所以实际的正则表达式里会看到两个反斜杠 \\ 表示一个反斜杠。
                    //\\s*：这匹配任意数量（包括零个）的空白字符（空格、制表符等）。这里使用两次，分别表示 firstUrl 和冒号 : 之间可能存在的空白字符，以及冒号和值之间的空白字符。\\s 是匹配单个空白字符，而 * 表示匹配前面元素的零次或多次出现。
                    //:：直接匹配一个冒号字符。
                    //\"(.*?)\"：这部分用于捕获 firstUrl 的值。\" 匹配双引号，(.*?) 中括号内的点 . 匹配除了换行符以外的任何单个字符，问号 ? 表示非贪婪模式，也就是尽可能少地匹配字符，直到遇到下一个双引号为止。圆括号 () 将匹配的内容标记为一个组，可以后续通过 matcher.group(1) 获取该组内容。
                    // 整体来看，这个正则表达式试图从脚本内容中找到形如 "firstUrl": "URL_VALUE" 的字符串，并提取出 URL_VALUE 部分作为结果返回。
                    Pattern pattern = Pattern.compile("\"firstUrl\"\\s*:\\s*\"(.*?)\"");
                    Matcher matcher = pattern.matcher(scriptContent);
                    if (matcher.find()) {
                        String firstUrl = matcher.group(1);
                        // 处理转义字符
                        firstUrl = firstUrl.replace("\\/", "/");
                        return firstUrl;
                    }
                }
            }

            throw new BusinessException(ErrorCode.OPERATION_ERROR.getCode(), "未找到 url");
        } catch (Exception e) {
            log.error("搜索失败", e);
            throw new BusinessException(ErrorCode.OPERATION_ERROR.getCode(), "搜索失败");
        }
    }

    public static void main(String[] args) {
        // 请求目标 URL
        String url = "https://graph.baidu.com/s?card_key=&entrance=GENERAL&extUiData[isLogoShow]=1&f=all&isLogoShow=1&session_id=3410322309181374013&sign=126c3e97cd54acd88139901737632149&tpl_from=pc";
        String imageFirstUrl = getImageFirstUrl(url);
        System.out.println("搜索成功，结果 URL：" + imageFirstUrl);
    }
}
